from transformers import pipeline

# Use a pipeline as a high-level helper
pipe = pipeline("text-classification", model="./chatgpt-detector-roberta-chinese")


def detect_ai_text(text):
    result = pipe(text)
    return result[0]["label"]


if __name__ == "__main__":
    human_text = "对于OpenAI大力出奇迹的工作，自然每个人都有自己的看点。我自己最欣赏的地方是ChatGPT如何解决 “AI校正(Alignment)“这个问题。这个问题也是我们课题组这两年在探索的学术问题之一。"

    chatgpt_text = "在您的情况下，考虑到事后已经过去12小时，并且考虑到避孕套破裂的情况，是建议尽快服用紧急避孕药的。紧急避孕药应在无防护性交后的72小时（3天）内服用，而且越早服用效果越好。从性交后的12小时内服用，避孕效果会最佳，随着时间的推移，紧急避孕药的有效性会逐渐减弱。您可以在不需要处方的情况下，前往药店或初级保健中心购买紧急避孕药。如果您服药后在3小时内出现呕吐或腹泻，应咨询医生。此外，即使服用了紧急避孕药，也不能保证一定能防止妊娠，因此如果在预期时间内没有来月经，建议做妊娠测试。请注意，紧急避孕药可能有一些副作用，如头痛、胃痛、恶心、呕吐、痛经和头晕。如有任何不适或疑问，应及时咨询医生或护士。"

    result = pipe([human_text, chatgpt_text])

    print(result)
